add back in problem 1 from home computer
add stad header
prettifying code