- Code clean up and added comments