Cleaned up code syntax