removed display of initial state (variables)
changed return type of method signatures to Unit
clean up: rearranging methods and renaming variables