Updated library code after visual testing issues fixed