Added final touches! Could refactor...maybe.