modified the file to standard python format
added a unit test for user