made it simpler
added setup and acknowledgements
shortened codes