cleaned and made better comments