Problem 1 and 2 of Assignment 2
Added PS2 Problem 3
Added PS2 Problem 4, and added assertion for the correct answer