#3. bug fixes