Adding code for first exercise