Improving my code post