Cleaned up code for easier reading