cleaned up code a tiny bit