Added code for bug deliverable one
Bug code from first deliverable