Basic app working. Is not the most beautiful code. Need a little refactoring