Improce code