Fixed minor styling issues and added import directive