code bit better