cleaned up code
added classList toggle