Added comments and removed leftover code from testing