Added a few lines to use the swtich