Slightly working prototype in the console - Working on cleaning code