Refactor to be generic, but Django only right now