fix(style): cleaned up code