First version of assignment. Need to add comments to code