add shorter solution