I forgot to add it to the Test and the main class again