Modified my assignment and re-wrote it.