Refactor function test
Trying to use South for migrations