cleaned up 2 lines of code