small changes after debugging
clean up code