solution changes

forgot to save the solution before committing last tiime..