Milestone 2

Added functionality for enter button