Added comments and cleaned up code
Added to the README