Ton of bug fixes