Slight changes in code
Slight formatting changes: