A little assignment 2