Started to implement if/else. Stripped for and while functionality