cleaned up code according to puppet style guide