Improve spacing in Stefan's assignment 1