sorted things
added needed package
implemented vince's comments
updates
work in progress