cleanup

clean code