Clean code

added comments