update to work with django