fixed some issues with the model, added some more variation