2nd challenge and changing into methods