worked on proj1, stage a, stage b