Day1 assignment done

This one is also right. But there is a shorter version than this which is one line and coool. :)