Code cleaning, some comments and stuff in README