Fixed basic methods and changed the main file to only accept an argument of code