Adding models directory, and some super small changes