This could be achieved in 7 lines of code.