Made changes to complete code