Hw 2 works need to clean it up