Working on switch case logic