Added bonus functionality