Use case statement; cleaner