Moved shot model to its own class