Add function to take input
Add checks for valid input
Clean up commments