Add switch statement
Update TODO