Remove decorators snapshot_create snapshot_restore. Replcae them with function prepare_environment.