First homework with better organization