lets put this in a class and run this b in code climate