Added code I am not proud of