The puzzle I solved to get an interview, I wanna improve it