Fixed type in homework 2