continued to edit the code and make it pretty