Updating Assignment 1

Added some buttons, logic.