Just some slight changes to the code

Made it such that I don't have to touch main.js to test my stuff