Added code and updated readme to include instructions