Add to_param method and cleanup code