A coding challenge I did for a code interview