V0.5.1

Cleaned up comments