Fixing code syntax