Added part of assignment 4