Added constructor to models