Finally got it working :/

Something like that?
Expected '{' and instead saw 'return'.