Added comments

Places were commented, and code was cleaned