Code update

Cleaned up code comments