Added problem4 code. Still working on it.