Restructured and updated code. Need to link code together and add documentation.