Version 0.2

Code improvements made and comments added to most classes