added model folder for better structure of the code