Updated script for django 1.7