Fixing based on comments

very nice. this is perfect