Hi, it passes the test but it feels like it can be done in a much better and shorter way... Any suggestions?