remove else part, add return instead