Cleaning up the code, adding functionality