modified div class
cleaned up code
cleaned up code