Pretty sure I got it working like that