Almost completely rewritten for cleaner code.