Working out return statements