Almost done

Just need to add one last thing and its done